from sdk.data import Data
from sdk.pic import Pic

if __name__ == '__main__':
  # SQL创建数据库
  # Data().createData()
  # Data().close()

  # CSV文件创建数据库
  Data().csvData()

  # 获取所有有奖金的员工信息
  sql = 'select ename,comm from emp where (comm is NOT NULL)'
  rows = Data().getDataBySql(sql)
  print('获取所有有奖金的员工信息数据',rows)
  Pic(data=rows,title='获取所有有奖金的员工信息').draw('获取所有有奖金的员工信息')

  # 查询各部门的最高薪水、最低薪水、平均薪水….
  sql2 = "SELECT b.loc,MAX(SAL),MIN(SAL),AVG(SAL) FROM EMP a,dept b where a.deptno=b.deptno GROUP BY b.loc;"
  rows = Data().getDataBySql(sql2)
  print('查询各部门的最高薪水、最低薪水、平均薪水',rows)
  
  data = [['header','最高薪水','最低薪水','平均薪水']]
  for x in rows:
    print(x)
    data.append(x)
  Pic(data=data,title='查询各部门的最高薪水、最低薪水、平均薪水').draw_series('查询各部门的最高薪水、最低薪水、平均薪水')